how to turn string to math function python